package initialize

import (
	_ "github.com/go-sql-driver/mysql"
	"go.uber.org/zap"
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"mcp-demo/internal/config/cmodel"
	"mcp-demo/internal/config/initialize/internal"
	"mcp-demo/internal/global"
)

// GormDoris 初始化Mysql数据库
func GormDoris() *gorm.DB {
	m := global.CONFIG.Doris
	return initDorisDatabase(m)
}

func CloseDoris(db *gorm.DB) {
	d, err := db.DB()
	if err != nil {
		return
	}
	_ = d.Close()
	zap.L().Info("Doris数据库已关闭")
}

// GormDorisByConfig 通过传入配置初始化Mysql数据库
func GormDorisByConfig(m cmodel.Doris) *gorm.DB {
	return initDorisDatabase(m)
}

// initDorisDatabase 初始化Doris数据库的辅助函数
func initDorisDatabase(m cmodel.Doris) *gorm.DB {
	if m.Dbname == "" {
		return nil
	}

	dorisConfig := mysql.Config{
		DSN:                       m.Dsn(), // DSN data source name
		DefaultStringSize:         191,     // string 类型字段的默认长度
		SkipInitializeWithVersion: false,   // 根据版本自动配置
	}

	if db, err := gorm.Open(mysql.New(dorisConfig), internal.Gorm.Config(m.Prefix, m.Singular, global.CONFIG.Doris.GeneralDB)); err != nil {
		panic(err)
	} else {
		db.InstanceSet("gorm:table_options", "ENGINE="+m.Engine)
		sqlDB, _ := db.DB()
		sqlDB.SetMaxIdleConns(m.MaxIdleConns)
		sqlDB.SetMaxOpenConns(m.MaxOpenConns)
		return db
	}
}
